home *** CD-ROM | disk | FTP | other *** search
/ Windows 95 API Bible / Windows 95 API Bible 3 Disc Set.iso / Win32 API Bible Book 3 of 3.iso / chapter4 / forkeys.c < prev    next >
C/C++ Source or Header  |  1996-03-21  |  11KB  |  304 lines
  1.  
  2. #include <windows.h>  
  3. #include <stdio.h>
  4. #include "forkeys.h"
  5. #include "sqlext.h"  
  6.  
  7. #if defined (WIN32)
  8.     #define IS_WIN32 TRUE
  9. #else
  10.     #define IS_WIN32 FALSE
  11. #endif
  12.  
  13. #define IS_NT      IS_WIN32 && (BOOL)(GetVersion() < 0x80000000)
  14. #define IS_WIN32S  IS_WIN32 && (BOOL)(!(IS_NT) && (LOBYTE(LOWORD(GetVersion()))<4))
  15. #define IS_WIN95   (BOOL)(!(IS_NT) && !(IS_WIN32S)) && IS_WIN32
  16.  
  17. HINSTANCE hInst;   // current instance
  18.  
  19. LPCTSTR lpszAppName = "MyApp";
  20. LPCTSTR lpszTitle   = "SQLForeignKeys()"; 
  21.  
  22.  
  23. BOOL RegisterWin95( CONST WNDCLASS* lpwc );
  24.  
  25.  
  26. int APIENTRY WinMain( HINSTANCE hInstance, HINSTANCE hPrevInstance,
  27.                       LPTSTR lpCmdLine, int nCmdShow)
  28. {
  29.    MSG      msg;
  30.    HWND     hWnd; 
  31.    WNDCLASS wc;
  32.  
  33.    wc.style         = CS_HREDRAW | CS_VREDRAW;
  34.    wc.lpfnWndProc   = (WNDPROC)WndProc;       
  35.    wc.cbClsExtra    = 0;                      
  36.    wc.cbWndExtra    = 0;                      
  37.    wc.hInstance     = hInstance;              
  38.    wc.hIcon         = LoadIcon (hInstance, lpszAppName); 
  39.    wc.hCursor       = LoadCursor(NULL, IDC_ARROW);
  40.    wc.hbrBackground = (HBRUSH)(COLOR_WINDOW+1);
  41.    wc.lpszMenuName  = lpszAppName;              
  42.    wc.lpszClassName = lpszAppName;              
  43.  
  44.    if ( IS_WIN95 )
  45.    {
  46.       if ( !RegisterWin95( &wc ) )
  47.          return( FALSE );
  48.    }
  49.    else if ( !RegisterClass( &wc ) )
  50.       return( FALSE );
  51.  
  52.    hInst = hInstance; 
  53.  
  54.    hWnd = CreateWindow( lpszAppName, 
  55.                         lpszTitle,    
  56.                         WS_OVERLAPPEDWINDOW, 
  57.                         CW_USEDEFAULT, 0, 
  58.                         CW_USEDEFAULT, 0,  
  59.                         NULL,              
  60.                         NULL,              
  61.                         hInstance,         
  62.                         NULL               
  63.                       );
  64.  
  65.    if ( !hWnd ) 
  66.       return( FALSE );
  67.  
  68.    ShowWindow( hWnd, nCmdShow ); 
  69.    UpdateWindow( hWnd );         
  70.  
  71.    while( GetMessage( &msg, NULL, 0, 0) )   
  72.    {
  73.       TranslateMessage( &msg ); 
  74.       DispatchMessage( &msg );  
  75.    }
  76.  
  77.    return( msg.wParam ); 
  78. }
  79.  
  80.  
  81. BOOL RegisterWin95( CONST WNDCLASS* lpwc )
  82. {
  83.    WNDCLASSEX wcex;
  84.  
  85.    wcex.style         = lpwc->style;
  86.    wcex.lpfnWndProc   = lpwc->lpfnWndProc;
  87.    wcex.cbClsExtra    = lpwc->cbClsExtra;
  88.    wcex.cbWndExtra    = lpwc->cbWndExtra;
  89.    wcex.hInstance     = lpwc->hInstance;
  90.    wcex.hIcon         = lpwc->hIcon;
  91.    wcex.hCursor       = lpwc->hCursor;
  92.    wcex.hbrBackground = lpwc->hbrBackground;
  93.    wcex.lpszMenuName  = lpwc->lpszMenuName;
  94.    wcex.lpszClassName = lpwc->lpszClassName;
  95.  
  96.    // Added elements for Windows 95.
  97.    //...............................
  98.    wcex.cbSize = sizeof(WNDCLASSEX);
  99.    wcex.hIconSm = LoadImage(wcex.hInstance, lpwc->lpszClassName, 
  100.                             IMAGE_ICON, 16, 16,
  101.                             LR_DEFAULTCOLOR );
  102.             
  103.    return RegisterClassEx( &wcex );
  104. }
  105.  
  106.  
  107. LRESULT CALLBACK WndProc( HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam )
  108. {
  109. static HENV  hEnv  = NULL;
  110. static HDBC  hDBC  = NULL;
  111. static HWND  hList = NULL;
  112.  
  113.    switch( uMsg )
  114.    {
  115.       case WM_CREATE :
  116.               {
  117.                  RETCODE retCode;
  118.  
  119.                  // Allocate Environment and Connection.
  120.                  //.....................................
  121.                  SQLAllocEnv( &hEnv );
  122.                  SQLAllocConnect( hEnv, &hDBC );
  123.  
  124.                  // Connect to the data source
  125.                  //...........................
  126.                  retCode = SQLConnect( hDBC, 
  127.                              "Test Data Source", SQL_NTS,
  128.                              "DBA",              SQL_NTS,  
  129.                              "SQL",              SQL_NTS );
  130.  
  131.                  if ( retCode != SQL_SUCCESS )
  132.                  {
  133.                     SQLFreeConnect( hDBC );
  134.                     SQLFreeEnv( hEnv );
  135.                     return( -1 );
  136.                  }
  137.  
  138.                  if ( retCode == SQL_SUCCESS )
  139.                  {
  140.                     int nTabSize = 60;
  141.  
  142.                     hList = CreateWindow( "LISTBOX", "",    
  143.                                LBS_NOTIFY | LBS_USETABSTOPS |
  144.                                LBS_NOINTEGRALHEIGHT  | LBS_SORT | 
  145.                                WS_BORDER  | WS_CHILD | 
  146.                                WS_VISIBLE | WS_VSCROLL, 
  147.                                0, 0, 0, 0, 
  148.                                hWnd, (HMENU)101,
  149.                                hInst, NULL );
  150.                     
  151.                     SendMessage( hList, LB_SETTABSTOPS, 
  152.                                  1, (LPARAM)&nTabSize );
  153.                  }
  154.               }
  155.               break;
  156.  
  157.       case WM_SIZE :
  158.               MoveWindow( hList, 0, 0, LOWORD( lParam ), 
  159.                                        HIWORD( lParam ), TRUE );
  160.               break; 
  161.               
  162.       case WM_COMMAND :
  163.               switch( LOWORD( wParam ) )
  164.               {
  165.                  case IDM_TEST :
  166.                         {
  167.                            HSTMT   hStmt;
  168.                            UCHAR   szPKTableName [128];
  169.                            UCHAR   szPKColumnName[128];
  170.                            UCHAR   szFKTableName [128];
  171.                            UCHAR   szFKColumnName[128];
  172.                            UCHAR   szLBStr       [128];
  173.                            SDWORD  dwTypeLen;
  174.                            RETCODE retCode;
  175.  
  176.                            // Allocate a statement handle
  177.                            // to receive the result set.
  178.                            // ...........................
  179.                            SQLAllocStmt( hDBC, &hStmt );
  180.  
  181.                            // Call SQLForeignKeys() to determine 
  182.                            // the foreign keys within the 
  183.                            // 'department' table and the columns
  184.                            // that the foreign keys refer to.
  185.                            // ..................................
  186.                            retCode = SQLForeignKeys( hStmt, 
  187.                                         NULL,         SQL_NTS, 
  188.                                         NULL,         SQL_NTS, 
  189.                                         NULL,         SQL_NTS,
  190.                                         NULL,         SQL_NTS,
  191.                                         NULL,         SQL_NTS,
  192.                                         "department", SQL_NTS );
  193.  
  194.                            if( retCode == SQL_SUCCESS || 
  195.                                retCode == SQL_SUCCESS_WITH_INFO )
  196.                            {
  197.                               // Bind a column to the PKTABLE_NAME, 
  198.                               // PKCOLUMN_NAME, FKTABLE_NAME, and
  199.                               // FKCOLUMN_NAME columns in the result
  200.                               // set generated by SQLForeignKeys().
  201.                               // ...................................
  202.                               SQLBindCol( hStmt, 3, SQL_C_CHAR, 
  203.                                  szPKTableName, sizeof( szPKTableName ), 
  204.                                  &dwTypeLen );
  205.  
  206.                               SQLBindCol( hStmt, 4, SQL_C_CHAR, 
  207.                                  szPKColumnName, sizeof( szPKColumnName ),
  208.                                  &dwTypeLen );
  209.  
  210.                               SQLBindCol( hStmt, 7, SQL_C_CHAR, 
  211.                                  szFKTableName, sizeof( szFKTableName ),
  212.                                  &dwTypeLen );
  213.  
  214.                               SQLBindCol( hStmt, 8, SQL_C_CHAR, 
  215.                                  szFKColumnName, sizeof( szFKColumnName ), 
  216.                                  &dwTypeLen );
  217.  
  218.                               // Fetch all the records and
  219.                               // display them in the list box.
  220.                               //..............................
  221.                               retCode = SQLFetch( hStmt );
  222.  
  223.                               while ( retCode == SQL_SUCCESS )
  224.                               {
  225.                                  // Set string contents to be  
  226.                                  // primary key table name, 
  227.                                  // primary key column name, 
  228.                                  // foreign key table name, and 
  229.                                  // foreign key column name,.
  230.                                  // ...........................
  231.                                  szLBStr[0] = '\0';
  232.                                  strcat( szLBStr, szPKTableName );
  233.                                  strcat( szLBStr, "\t" );
  234.                                  strcat( szLBStr, szPKColumnName );
  235.                                  strcat( szLBStr, "\t" );
  236.                                  strcat( szLBStr, szFKTableName );
  237.                                  strcat( szLBStr, "\t" );
  238.                                  strcat( szLBStr, szFKColumnName );
  239.  
  240.                                  SendMessage( hList, LB_ADDSTRING, 
  241.                                               0, (LPARAM)szLBStr );
  242.  
  243.                                  retCode = SQLFetch( hStmt );
  244.                               }
  245.                            }
  246.  
  247.                            SQLFreeStmt( hStmt, SQL_DROP );
  248.                         }
  249.                         break;
  250.  
  251.                  case IDM_ABOUT :
  252.                         DialogBox( hInst, "AboutBox", hWnd, (DLGPROC)About );
  253.                         break;
  254.  
  255.                  case IDM_EXIT :
  256.                         DestroyWindow( hWnd );
  257.                         break;
  258.               }
  259.               break;
  260.       
  261.       case WM_DESTROY :
  262.               PostQuitMessage(0);
  263.  
  264.               // Disconnect Environment.
  265.               //........................
  266.               SQLDisconnect( hDBC );
  267.  
  268.               // Free Connection and Environment.
  269.               //.................................
  270.               SQLFreeConnect( hDBC );
  271.               SQLFreeEnv( hEnv );
  272.               break;
  273.  
  274.       default :
  275.             return( DefWindowProc( hWnd, uMsg, wParam, lParam ) );
  276.    }
  277.  
  278.    return( 0L );
  279. }
  280.  
  281.  
  282. LRESULT CALLBACK About( HWND hDlg,           
  283.                         UINT message,        
  284.                         WPARAM wParam,       
  285.                         LPARAM lParam)
  286. {
  287.    switch (message) 
  288.    {
  289.        case WM_INITDIALOG: 
  290.                return (TRUE);
  291.  
  292.        case WM_COMMAND:                              
  293.                if (   LOWORD(wParam) == IDOK         
  294.                    || LOWORD(wParam) == IDCANCEL)    
  295.                {
  296.                        EndDialog(hDlg, TRUE);        
  297.                        return (TRUE);
  298.                }
  299.                break;
  300.    }
  301.  
  302.    return (FALSE); 
  303. }
  304.